package TabelModels;

import java.util.ArrayList;

import javax.swing.table.AbstractTableModel;

import model.Dier;
import Lijsten.PelikaanLijst;

@SuppressWarnings("serial")
public class PelikaanTabelModel extends AbstractTableModel{
	
	private String[] names = {"naam", "vacht", "humeur", "blaas", "ziekte", "geslacht", "zwanger"};
	private ArrayList<Dier> dieren;
	
	
	public PelikaanTabelModel(String geslacht){

		if(geslacht.equals("Man")){
			
			dieren = PelikaanLijst.getInstance().getMannetjes();
		}
		else if(geslacht.equals("Vrouw")){
			
			dieren = PelikaanLijst.getInstance().getVrouwtjes();
		}
		else{
			
		}
	}


	@Override
	public int getColumnCount() {
	
		return 7;
	}


	@Override
	public int getRowCount() {

		return dieren.size();
	}

	public String getColumnName(int arg0) {
		
		return names[arg0];
	}

	@Override
	public Object getValueAt(int arg0, int arg1) {
		Dier aap = dieren.get(arg0);
		if(arg1 == 0){
			return aap.getNaam();
		}
		else if(arg1 == 1){
			return aap.getVacht();
		}
		else if(arg1 == 2){
			return aap.getHumeur();
		}
		else if(arg1 == 3){
			return aap.isBlaas();
		}
		else if(arg1 == 4){
			return aap.isZiek();
		}
		else if(arg1 == 5){
			return aap.isMannelijk();
		}
		else if(arg1 == 6){
			return aap.isZwanger();
		}
		
		return null;
	}
	
	public Dier getDier(int i){
		return dieren.get(i);
	}

}
